from pydantic import BaseModel
from typing import List, Dict
from openai import Agent

class SSDSL(BaseModel):
    diagram_name: str
    lifelines: List[str]
    messages: List[Dict[str, str]]

agent_ssd = Agent(
    name="SSDGenerator",
    instructions="基于需求描述生成系统顺序图DSL。",
    output_type=SSDSL,
    model="gpt-4o"
)